/*
 * Copyright (c) 2011.
 *     This program is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     This program is distributed in the hope that it will be useful,
 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *     GNU General Public License for more details.
 *
 *     You should have received a copy of the GNU General Public License
 *     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.graphoman.undo;

import org.graphoman.model.Model;
import org.graphoman.model.State;
import org.graphoman.model.Transition;

/**
 * Created by IntelliJ IDEA.
 * User: daniil
 * Date: 17.10.11
 * Time: 10:12
 * To change this template use File | Settings | File Templates.
 */
public class CreateAction implements UndoAction{
    Object newObject;

    public CreateAction(Object newObject) {
        this.newObject = newObject;
    }

    public void undo(Model model) {
        // delete object we have created
        if (newObject instanceof State) {
            model.states.remove(newObject);
        } else if (newObject instanceof Transition) {
            model.transitions.remove(newObject);
        }
    }
}
